#!/usr/bin/python
# -*- coding: UTF-8 -*-
import sys,getopt
import pandas
import re

def main(argv):
	print("hello world")
	print("hello world")
	inlist = ''
	### 1.参数判断
	opts,args = getopt.getopt(argv,"hi:",["help","infile="])
	for opt,arg in opts:
		if opt in ('-h',"--help"):
			print('merge_cxcel.py -i <list>')
			sys.exit()
		elif opt in ("-i","--infile"):
			inlist = arg
		
	print('输入list为 ：',inlist)
#	sys.exit()
	
	#i=1
	pattern_wh=r"/\w\w\wwh\d/"
	pattern_sz=r"/\w\w\wsz\d/"
	clean_array=[]
	### 2.读取list文件，对每一个excel进行操作
	file_obj=open(inlist,'r',encoding='UTF-8')
	for line in file_obj.readlines():
		line = line.strip()
		print(line)

		reader = pandas.ExcelFile(line)
		sheet_names = reader.sheet_names
		for sheet in sheet_names:
			print("\t"+sheet)
			raw_sheet_data = pandas.read_excel(line,sheet_name=sheet)

			if "只认领保留的，删除的不用反馈" in raw_sheet_data.columns:
				clean_sheet_data = raw_sheet_data.loc[raw_sheet_data["只认领保留的，删除的不用反馈"] == "保留"]
			else:
				continue
#			sys.exit()
			if clean_sheet_data.empty:
				continue
			else:
				clean_array.append(clean_sheet_data)
	
#				for index in clean_sheet_data.index :
#					if re.match(pattern_wh,clean_sheet_data.iat[index,0]):
#						wh_dict = obj2dict(clean_sheet_data.iloc[index])
#						df = pandas.DataFrame(wh_dict)
#						print(df)
#						sys.exit()
#						print("\t\twh")
#					elif re.match(pattern_sz,clean_sheet_data.iat[index,0]):
#						sz_dict = obj2dict(clean_sheet_data.iloc[index])
#						df = pandas.DataFrame(sz_dict)
#						print("\t\tsz")
			
#			print(wh_array)
#			sys.exit()

	file_obj.close()

	### 3.合并
#	print(clean_array)
	all_clean=pandas.concat(clean_array)
	all_clean.to_excel("all.xlsx")


if __name__ == "__main__":
	main(sys.argv[1:])


### 读取excel文件
## 需要解决的问题，怎么获得excel的sheet数；
## 讲获取的excel数据转换为字典
## 各种意外情况是否可以获取到数据
# 隐藏行，隐藏列，多个sheet


### 整合信息

### 写入excel文件